#!/bin/bash
# ============================================================
# 脚本名称：comfyui_restore_models.sh
# 功能说明：将已打包的模型 tar.gz 文件解压还原到指定 ComfyUI 的 models 目录
#
# 使用示例：
#   bash comfyui_restore_models.sh /workspace/comfyui/ComfyUI comfyui_models.tar.gz
#
# 参数说明：
#   $1 - ComfyUI 根目录路径（可选，默认：/workspace/comfyui/ComfyUI）
#   $2 - 模型 tar.gz 压缩包文件路径（可选，默认：comfyui_models_backup.tar.gz）
# ============================================================

COMFYUI_ROOT=${1:-/workspace/comfyui/ComfyUI}
ARCHIVE_PATH=${2:-comfyui_models_backup.tar.gz}

echo "📦 正在解压模型归档文件..."
echo "📁 目标 ComfyUI 路径: $COMFYUI_ROOT"
echo "📜 模型归档文件:     $ARCHIVE_PATH"

if [ ! -f "$ARCHIVE_PATH" ]; then
    echo "❌ 错误：找不到归档文件 $ARCHIVE_PATH"
    exit 1
fi

tar -xzvf "$ARCHIVE_PATH" -C "$COMFYUI_ROOT"

echo "✅ 模型已还原至：$COMFYUI_ROOT/models"
